package com.liu.ai.prompt;

import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.ai.chat.prompt.PromptTemplate;

import java.util.Map;

public class PromptDemo {

    public static void main(String[] args) {

//        String promptWithContext = """
//                用户问题：{question}
//                下面是上下文信息
//                ---------------------
//                {content}
//                ---------------------
//                给定的上下文和提供的历史信息，而不是事先的知识，回复用户的意见。
//                如果答案不在上下文中，告诉用户你不能回答这个问题。
//                限制在100个字内。
//                """;
//
//
//        PromptTemplate promptTemplate = new PromptTemplate(
//                promptWithContext,
//                Map.of("question","我是问题","content","我是上下问内容"));
//        String render = promptTemplate.render();
//        Prompt prompt = new Prompt(render);
//
//        System.out.println(prompt);




        //提示词模板
        String promptContext = """
                理论知识： {dataset}
                真实数据： {realData}
                回答要求： {prompt}
                """;
        Map<String, Object> map = Map.of("realData", "真实数据", "dataset", "理论知识", "prompt", "提示词");
        String messagePrompt = new PromptTemplate(promptContext,map).render();
        System.out.println( messagePrompt);
    }
}
